home *** CD-ROM | disk | FTP | other *** search
/ MacFormat 1995 July / macformat-026.iso / mac / Shareware City / Developers / berkeleydb1.73 / Berkeley_db / recno / rec_search.c < prev    next >
Encoding:
C/C++ Source or Header  |  1994-07-27  |  4.0 KB  |  132 lines  |  [TEXT/MPS ]
  1. /*-
  2.  * Copyright (c) 1990, 1993
  3.  *    The Regents of the University of California.  All rights reserved.
  4.  *
  5.  * Redistribution and use in source and binary forms, with or without
  6.  * modification, are permitted provided that the following conditions
  7.  * are met:
  8.  * 1. Redistributions of source code must retain the above copyright
  9.  *    notice, this list of conditions and the following disclaimer.
  10.  * 2. Redistributions in binary form must reproduce the above copyright
  11.  *    notice, this list of conditions and the following disclaimer in the
  12.  *    documentation and/or other materials provided with the distribution.
  13.  * 3. All advertising materials mentioning features or use of this software
  14.  *    must display the following acknowledgement:
  15.  *    This product includes software developed by the University of
  16.  *    California, Berkeley and its contributors.
  17.  * 4. Neither the name of the University nor the names of its contributors
  18.  *    may be used to endorse or promote products derived from this software
  19.  *    without specific prior written permission.
  20.  *
  21.  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24.  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31.  * SUCH DAMAGE.
  32.  */
  33.  
  34. #if defined(LIBC_SCCS) && !defined(lint)
  35. static char sccsid[] = "@(#)rec_search.c    8.2 (Berkeley) 9/14/93";
  36. #endif /* LIBC_SCCS and not lint */
  37.  
  38. #if defined(macintosh) && (defined(powerc) || defined(__powerc))
  39. #include "OurMalloc.h"
  40. #endif
  41.  
  42. #include <sys/types.h>
  43.  
  44. #include <errno.h>
  45. #include <stdio.h>
  46.  
  47. #include <db.h>
  48. #include "recno.h"
  49.  
  50. /*
  51.  * __REC_SEARCH -- Search a btree for a key.
  52.  *
  53.  * Parameters:
  54.  *    t:    tree to search
  55.  *    recno:    key to find
  56.  *    op:     search operation
  57.  *
  58.  * Returns:
  59.  *    EPG for matching record, if any, or the EPG for the location of the
  60.  *    key, if it were inserted into the tree.
  61.  *
  62.  * Returns:
  63.  *    The EPG for matching record, if any, or the EPG for the location
  64.  *    of the key, if it were inserted into the tree, is entered into
  65.  *    the bt_cur field of the tree.  A pointer to the field is returned.
  66.  */
  67. EPG *
  68. __rec_search(t, recno, op)
  69.     BTREE *t;
  70.     recno_t recno;
  71.     enum SRCHOP op;
  72. {
  73.     register indx_t index;
  74.     register PAGE *h;
  75.     EPGNO *parent;
  76.     RINTERNAL *r;
  77.     pgno_t pg;
  78.     indx_t top;
  79.     recno_t total;
  80.     int serrno;
  81.  
  82.     BT_CLR(t);
  83.     for (pg = P_ROOT, total = 0;;) {
  84.         if ((h = mpool_get(t->bt_mp, pg, 0)) == NULL)
  85.             goto err;
  86.         if (h->flags & P_RLEAF) {
  87.             t->bt_cur.page = h;
  88.             t->bt_cur.index = recno - total;
  89.             return (&t->bt_cur);
  90.         }
  91.         for (index = 0, top = NEXTINDEX(h);;) {
  92.             r = GETRINTERNAL(h, index);
  93.             if (++index == top || total + r->nrecs > recno)
  94.                 break;
  95.             total += r->nrecs;
  96.         }
  97.  
  98.         if (__bt_push(t, pg, index - 1) == RET_ERROR)
  99.             return (NULL);
  100.         
  101.         pg = r->pgno;
  102.         switch (op) {
  103.         case SDELETE:
  104.             --GETRINTERNAL(h, (index - 1))->nrecs;
  105.             mpool_put(t->bt_mp, h, MPOOL_DIRTY);
  106.             break;
  107.         case SINSERT:
  108.             ++GETRINTERNAL(h, (index - 1))->nrecs;
  109.             mpool_put(t->bt_mp, h, MPOOL_DIRTY);
  110.             break;
  111.         case SEARCH:
  112.             mpool_put(t->bt_mp, h, 0);
  113.             break;
  114.         }
  115.  
  116.     }
  117.     /* Try and recover the tree. */
  118. err:    serrno = errno;
  119.     if (op != SEARCH)
  120.         while  ((parent = BT_POP(t)) != NULL) {
  121.             if ((h = mpool_get(t->bt_mp, parent->pgno, 0)) == NULL)
  122.                 break;
  123.             if (op == SINSERT)
  124.                 --GETRINTERNAL(h, parent->index)->nrecs;
  125.             else
  126.                 ++GETRINTERNAL(h, parent->index)->nrecs;
  127.                         mpool_put(t->bt_mp, h, MPOOL_DIRTY);
  128.                 }
  129.     errno = serrno;
  130.     return (NULL);
  131. }
  132.